HarnessSQL: Harness-Native Training for SQL Agents in Realistic Database Environments
HarnessSQL:真实数据库环境中 SQL Agent 的 Harness 原生训练
🧭 SQL Agent 原生训练 · 执行奖励 RL
Hong Kong University of Science and Technology · Microsoft Research · Tsinghua University · University of Macau
提出 HarnessSQL 训练框架,在 SFT 和 RL 中保留完整 SQL 交互结构,大幅提升小模型在真实数据库任务中的执行准确率
📌 工作参考价值(4 点)
- 借鉴 harness-native 训练思想:在电商搜索/推荐 agent 训练时,让模型直接调用线上同构工具(商品检索 API、数据库查询、排序服务),保留完整交互轨迹做 SFT,而不是只输入问题输出答案,缩小训练与线上推理差距。
- 使用 execution-reward RL:以可验证的结果(如商品是否被正确召回、SQL 是否返回正确结果、用户是否点击)作为奖励信号,只对最终成功轨迹给予正奖励,能有效提升长程 agent 任务表现,且无需逐步骤标注。
- 引入 hidden oracle 自动过滤高质量轨迹:在业务中可构建规则或脚本验证任务完成情况,从海量交互日志中筛选成功样本用于训练,降低人工标注成本。
- 隔离可执行沙箱环境:在离线训练时为 agent 提供稳定、可控的数据库/服务沙箱,避免真实环境波动导致训练不稳定;同时可模拟错误注入,增强错误恢复能力。